import os
import sys
import pandas as pd
import datetime
from MainConsolidator import MainConsolidator

class handler(object):
    @staticmethod
    def module_name():
        return 'Лист, штрипс'
    @staticmethod
    def handle(row:pd.Series,consolidator:MainConsolidator)->pd.Series:
        product_up=None
        try:
            product = row["Первичный продукт"]
            strength = consolidator.strength_by_steel(row["Марка стали <Производство>"]) 
            if pd.isna(strength): strength="К52"
            strength_steel = consolidator.strength_mapping(strength,row["Марка стали <Производство>"])
            strength_led = consolidator.strength_led(strength_steel[0])
            if strength_led=="К0": strength_led="К52"
            gmsps = consolidator.get_gmsps(row["Марка стали <Производство>"],product)
            if "ГЮ" in gmsps:
                gmsps_ = consolidator.scalar_in_guid("Признак по НТД",
                    f"`Производство`=='Лист, штрипс' and `НТД качества (трубы стальные) <Производство>`=='{row['НТД качества (прокат) <Производство>']}'","Доп параметр1")
                if pd.notna(gmsps_): gmsps=gmsps_
            
            diametr_=0
            thickness=row["Толщина, мм <Производство>"]
            width = row["Ширина листа (штрипса), мм <Производство>"]
            print(f"strength_steel:{strength_steel[0]}; strength_led:{strength_led}; gmsps: {gmsps}")
            ## issue http://vsys01775:8282/flea/pyexcelcons3/-/issues/3
            if gmsps=="ГУ": 
                width_led = consolidator.nearest_in_guid("Параметры по Продукту Прокат",
                    f"`Производство`=='Лист. штрипс' and `Продукт`=='{product}' and `Мин`<={thickness}<=`Макс` and `Доп параметр2`=='ГУ'",
                    "Доп параметр1",width,"Доп параметр1")
            else:
                width_led = consolidator.nearest_in_guid("Параметры по Продукту Прокат",
                    f"`Производство`=='Лист. штрипс' and `Продукт`=='{product}' and `Доп параметр2`=='{gmsps}'",
                    "Доп параметр1",width,"Доп параметр1")                
            width_led_s=str(width_led).replace(".0","")
            
            
            if width==1500 or width==2000: diametr_=0
            q1020_1 = (width - 5) * (1 - 0.4 / 100) * (1 + 1.1 / 100) / 3.1416 + thickness
            q1420_1 = (width - 5) * (1 - 0 / 100) * (1 + 0.6 / 100) / 3.1416 + thickness
            d1020 = consolidator.nearest_in_guid("Диаметры ТБД","`Диаметры`>0","Диаметры",q1020_1,"Диаметры")
            d1420 = consolidator.nearest_in_guid("Диаметры ТБД","`Диаметры`>0","Диаметры",q1420_1,"Диаметры")            
            delta_abs_1020=abs(q1020_1-d1020)
            delta_abs_1420=abs(q1420_1-d1420)   
            if delta_abs_1020<delta_abs_1420 : diametr_=d1020
            else : diametr_=d1420          
                        
            if product=="ЛИСТ Р":
                pass
            if product=="ЛИСТ ТРБ":
                pass
            if product=="ПОЛОСА":
                product_up = f"{product} Ш{width_led_s} {gmsps}"
            if pd.notna(row["Признак БЗ"]) and  row["Признак БЗ"]=="БЗ": product_up="ПОЛОСА Р БЗ"
            elif "ГУ" in gmsps:
                thickness_range = consolidator.scalar_in_guid("Параметры по Продукту Прокат",
                f"`Производство`=='Лист. штрипс' and `Продукт`=='{product}' and `Доп параметр2`=='ГУ' and `Доп параметр1`=={width_led}",
                "Диапазон")
                product_up=f"{product} {thickness_range} Ш{width_led_s} {gmsps}"           
            
            print(f"{product} ;  d1020: {q1020_1} => {d1020} => {delta_abs_1020}; d1420: {q1420_1} => {d1420} => {delta_abs_1420}; width_led: {width_led} ")
            
            row["error"] = pd.NA
        except Exception as exp: 
            row["error"] = f"{exp}"
        finally:
            row["Продукт УП"]=product_up
            row["Время укрупнения"]=datetime.datetime.now()
            return row
        
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
        try:
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
        except: raise
    
    def get_compiled(self):
        code:str=f"print('{__file__}')"
        try:
            with open(__file__,encoding="utf-8",mode="r") as f:
                code=f.read()
        except Exception as exp: 
            code=f"print('error: {exp}')"
        finally:
            cc = compile(code,"","exec")
            return cc

if __name__=="__sheet_strips__":
    consolidator:MainConsolidator
    record:pd.Series
    result:pd.Series
    result = handler.handle(record,consolidator)

